import random
import time
from datetime import datetime
from api.ihrm_emp_curd import IhrmEmpCURD
def generate_unique_mobile():
    """
    生成唯一的手机号码
    规则:138 + 8位随机数, 并在最后加入当前时间戳的后几位确保唯一性
    """
    # 基础前缀
    prefix = "138"
    # 生成5位随机数
    random_part = ''.join([str(random.randint(0, 9)) for _ in range(5)])
    # 获取当前时间戳的后4位作为唯一性保证
    timestamp = str(int(time.time()))[-4:]
    # 组合成11位手机号
    mobile = prefix + random_part + timestamp
    return mobile
def add_emp_with_unique_mobile(header, json_data):

    # 检查是否已提供手机号
    if not json_data.get("mobile"):
        # 生成唯一手机号
        unique_mobile = generate_unique_mobile()
        json_data["mobile"] = unique_mobile
        print(f"已自动生成唯一手机号: {unique_mobile}")
    # 调用原有的添加员工方法
    return IhrmEmpCURD.add_emp(header, json_data)